#-*- codeing -utf-8 -*-
#@Time : 2022/1/11 12:18
#@Author : 19-软件-林昊天
#@File : testCloud.py
#@Software: PyCharm

import jieba    #分词
from matplotlib import pyplot as plt    #绘图，数据可视化
from wordcloud import WordCloud #词云
from PIL import Image   #图片处理
import numpy as np  #矩阵运算
import sqlite3

#准备词云所需文字或词
coon=sqlite3.connect('topmovies.db')
cur=coon.cursor()
sql="select introduction from movie250"
data=cur.execute(sql)
text=""
for item in data:
    text=text+item[0]
cur.close()
coon.close()

cut=jieba.cut(text)
string=' '.join(cut)
print(len(string))

img=Image.open("tree.jpg")  #打开遮罩图片
img_array=np.array(img) #将图片转换为数组
wc=WordCloud(
    background_color='white',
    mask=img_array,
    font_path="FZSTK.TTF",   #字体所在位置，C:\Windows\Fonts
    scale=3
)
wc.generate_from_text(string)

#绘制图片
fig=plt.figure(1)
plt.subplots(figsize=(12,8),dpi=300)    #使照片更清晰
plt.imshow(wc)
plt.axis('off') #是否显示坐标轴

# plt.show()  #显示生成的词云图片
#输出词云图片到文件
plt.savefig('static/assets/img/ciyun.jpg')
